package com.example.lcpractice.lc;

import com.example.lcpractice.datastructure.ListNode;

import java.util.HashSet;
import java.util.Set;

/**
 * [19]删除链表的倒数第 N 个结点.java
 * 给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。
 * <p>
 * 示例 1：
 * <p>
 * <p>
 * 输入：head = [1,2,3,4,5], n = 2
 * 输出：[1,2,3,5]
 * 示例 2：
 * <p>
 * 输入：head = [1], n = 1
 * 输出：[]
 * 示例 3：
 * <p>
 * 输入：head = [1,2], n = 1
 * 输出：[1]
 * 提示：
 * <p>
 * 链表中结点的数目为 sz
 * 1 <= sz <= 30
 * 0 <= Node.val <= 100
 * 1 <= n <= sz
 * 进阶：你能尝试使用一趟扫描实现吗？
 * <p>
 * Related Topics
 * 链表
 * 双指针
 *
 * @author hd
 * @date 2022-07-08 09:51
 * @description 回文数
 */
public class Lc19 {
    public static void main(String[] args) {
        ListNode listNode = new ListNode(1);
        ListNode listNode2 = new ListNode(2);
        ListNode listNod3 = new ListNode(3);
        listNode.setNext(listNode2);
        listNode2.setNext(listNod3);
        System.out.println(removeNthFromEnd(ListNode.getSingleLinkedList(), 1));
        System.out.println(removeNthFromEnd(new ListNode(1), 1));
        System.out.println(removeNthFromEnd(listNode, 3));
    }

    public static ListNode removeNthFromEnd(ListNode head, int n) {
        if (head.next == null) {
            return null;
        }
        ListNode fore = head;
        ListNode next = head.next;

        ListNode back = head;
        int count = 0;

        while (back.next != null) {
            back = back.next;
            count++;
            if (count > (n)) {
                fore = fore.next;
                next = next.next;
            }
        }
        if (n - count == 1) {
            return head.next;
        }
        fore.next = next.next;
        return head;
    }

    /**
     * 跑得快先指针
     * @param head
     * @param n
     * @return
     */
    public static ListNode removeNthFromEnd2(ListNode head, int n) {
        //删除后返回 pre.next，为什么不直接返回 head 呢，因为 head 有可能是被删掉的点
        //如同以上
        //比如3个的长度，删除倒数3
        // 如果不加占位 [1,2,3],n=3,  || 开始：start=2,n=2;start=3,n=1; start=null;n=0 空指针异常

        // 加占位符    [0,1,2,3],n=3  || start=1,n=2;start=2,n=1; start=3;n=0   null不执行 end[占位].next[1].next[2] 中间被挤掉
        ListNode pre = new ListNode(0);
        pre.next = head;
        ListNode start = pre, end = pre;
        //先跑n个，与end指针跑开间隔
        while (n != 0) {
            start = start.next;
            n--;
        }
        while (start.next != null) {
            start = start.next;
            end = end.next;
        }
        end.next = end.next.next;
        return pre.next;
    }
}

